import tensorflow as tf


tf.app.flags.DEFINE_string(flag_name="path",default_value='~/Desktop/',docstring="桌面的路径")

flg = tf.app.flags.FLAGS


def main(argv):
    print('in main')
    print(argv)
    # 获取自定义的命令行参数
    print(flg.path)

if __name__ == '__main__':
    tf.app.run()